IndexError: too many indices for array when trying to use scipy.optimize.minimize